import cv2
import numpy as np

# 加载图像
image = cv2.imread('input.jpg')

# 将图像转换为HSV色彩空间
hsv_image = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)

# 在HSV色彩空间中，设置一个范围来选择颜色
lower_color = np.array([0, 100, 100])  # 选择颜色范围的下限
upper_color = np.array([10, 255, 255])  # 选择颜色范围的上限
color_mask = cv2.inRange(hsv_image, lower_color, upper_color)

# 对图像进行颜色过滤
result = cv2.bitwise_and(image, image, mask=color_mask)

# 显示原始图像和过滤后的图像
cv2.imshow('Original Image', image)
cv2.imshow('Filtered Image', result)
cv2.waitKey(0)
cv2.destroyAllWindows()